home *** CD-ROM | disk | FTP | other *** search
/ CD Actual 53 / PC Actual CD 53.iso / Share / Progra / python / BeOpen-Python-2.0.exe / TEST_USERLIST.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  3.9 KB  |  201 lines
  1. # Check every path through every method of UserList
  2.  
  3. from UserList import UserList
  4. from test_support import TestFailed
  5.  
  6. # Use check instead of assert so -O doesn't render the
  7. # test useless.
  8. def check(predicate, msg):
  9.     if not predicate:
  10.         raise TestFailed(msg + " failed")
  11.  
  12. l0 = []
  13. l1 = [0]
  14. l2 = [0, 1]
  15.  
  16. # Test constructors
  17.  
  18. u = UserList()
  19. u0 = UserList(l0)
  20. u1 = UserList(l1)
  21. u2 = UserList(l2)
  22.  
  23. uu = UserList(u)
  24. uu0 = UserList(u0)
  25. uu1 = UserList(u1)
  26. uu2 = UserList(u2)
  27.  
  28. v = UserList(tuple(u))
  29. class OtherList:
  30.     def __init__(self, initlist):
  31.         self.__data = initlist
  32.     def __len__(self):
  33.         return len(self.__data)
  34.     def __getitem__(self, i):
  35.         return self.__data[i]
  36. v0 = UserList(OtherList(u0))
  37. vv = UserList("this is also a sequence")
  38.  
  39. # Test __repr__
  40.  
  41. check(str(u0) == str(l0), "str(u0) == str(l0)")
  42. check(repr(u1) == repr(l1), "repr(u1) == repr(l1)")
  43. check(`u2` == `l2`, "`u2` == `l2`")
  44.  
  45. # Test __cmp__ and __len__
  46.  
  47. def mycmp(a, b):
  48.     r = cmp(a, b)
  49.     if r < 0: return -1
  50.     if r > 0: return 1
  51.     return r
  52.  
  53. all = [l0, l1, l2, u, u0, u1, u2, uu, uu0, uu1, uu2]
  54. for a in all:
  55.     for b in all:
  56.         check(mycmp(a, b) == mycmp(len(a), len(b)),
  57.               "mycmp(a, b) == mycmp(len(a), len(b))")
  58.  
  59. # Test __getitem__
  60.  
  61. for i in range(len(u2)):
  62.     check(u2[i] == i, "u2[i] == i")
  63.  
  64. # Test __setitem__
  65.  
  66. uu2[0] = 0
  67. uu2[1] = 100
  68. try:
  69.     uu2[2] = 200
  70. except IndexError:
  71.     pass
  72. else:
  73.     raise TestFailed("uu2[2] shouldn't be assignable")
  74.  
  75. # Test __delitem__
  76.  
  77. del uu2[1]
  78. del uu2[0]
  79. try:
  80.     del uu2[0]
  81. except IndexError:
  82.     pass
  83. else:
  84.     raise TestFailed("uu2[0] shouldn't be deletable")
  85.  
  86. # Test __getslice__
  87.  
  88. for i in range(-3, 4):
  89.     check(u2[:i] == l2[:i], "u2[:i] == l2[:i]")
  90.     check(u2[i:] == l2[i:], "u2[i:] == l2[i:]")
  91.     for j in range(-3, 4):
  92.         check(u2[i:j] == l2[i:j], "u2[i:j] == l2[i:j]")
  93.  
  94. # Test __setslice__
  95.  
  96. for i in range(-3, 4):
  97.     u2[:i] = l2[:i]
  98.     check(u2 == l2, "u2 == l2")
  99.     u2[i:] = l2[i:]
  100.     check(u2 == l2, "u2 == l2")
  101.     for j in range(-3, 4):
  102.         u2[i:j] = l2[i:j]
  103.         check(u2 == l2, "u2 == l2")
  104.  
  105. uu2 = u2[:]
  106. uu2[:0] = [-2, -1]
  107. check(uu2 == [-2, -1, 0, 1], "uu2 == [-2, -1, 0, 1]")
  108. uu2[0:] = []
  109. check(uu2 == [], "uu2 == []")
  110.  
  111. # Test __contains__
  112. for i in u2:
  113.     check(i in u2, "i in u2")
  114. for i in min(u2)-1, max(u2)+1:
  115.     check(i not in u2, "i not in u2")
  116.  
  117. # Test __delslice__
  118.  
  119. uu2 = u2[:]
  120. del uu2[1:2]
  121. del uu2[0:1]
  122. check(uu2 == [], "uu2 == []")
  123.  
  124. uu2 = u2[:]
  125. del uu2[1:]
  126. del uu2[:1]
  127. check(uu2 == [], "uu2 == []")
  128.  
  129. # Test __add__, __radd__, __mul__ and __rmul__
  130.  
  131. check(u1 + [] == [] + u1 == u1, "u1 + [] == [] + u1 == u1")
  132. check(u1 + [1] == u2, "u1 + [1] == u2")
  133. check([-1] + u1 == [-1, 0], "[-1] + u1 == [-1, 0]")
  134. check(u2 == u2*1 == 1*u2, "u2 == u2*1 == 1*u2")
  135. check(u2+u2 == u2*2 == 2*u2, "u2+u2 == u2*2 == 2*u2")
  136. check(u2+u2+u2 == u2*3 == 3*u2, "u2+u2+u2 == u2*3 == 3*u2")
  137.  
  138. # Test append
  139.  
  140. u = u1[:]
  141. u.append(1)
  142. check(u == u2, "u == u2")
  143.  
  144. # Test insert
  145.  
  146. u = u2[:]
  147. u.insert(0, -1)
  148. check(u == [-1, 0, 1], "u == [-1, 0, 1]")
  149.  
  150. # Test pop
  151.  
  152. u = [-1] + u2
  153. u.pop()
  154. check(u == [-1, 0], "u == [-1, 0]")
  155. u.pop(0)
  156. check(u == [0], "u == [0]")
  157.  
  158. # Test remove
  159.  
  160. u = u2[:]
  161. u.remove(1)
  162. check(u == u1, "u == u1")
  163.  
  164. # Test count
  165. u = u2*3
  166. check(u.count(0) == 3, "u.count(0) == 3")
  167. check(u.count(1) == 3, "u.count(1) == 3")
  168. check(u.count(2) == 0, "u.count(2) == 0")
  169.  
  170.  
  171. # Test index
  172.  
  173. check(u2.index(0) == 0, "u2.index(0) == 0")
  174. check(u2.index(1) == 1, "u2.index(1) == 1")
  175. try:
  176.     u2.index(2)
  177. except ValueError:
  178.     pass
  179. else:
  180.     raise TestFailed("expected ValueError")
  181.  
  182. # Test reverse
  183.  
  184. u = u2[:]
  185. u.reverse()
  186. check(u == [1, 0], "u == [1, 0]")
  187. u.reverse()
  188. check(u == u2, "u == u2")
  189.  
  190. # Test sort
  191.  
  192. u = UserList([1, 0])
  193. u.sort()
  194. check(u == u2, "u == u2")
  195.  
  196. # Test extend
  197.  
  198. u = u1[:]
  199. u.extend(u2)
  200. check(u == u1 + u2, "u == u1 + u2")
  201.